<?php
/**
 * Zend Framework
 *
 * LICENSE
 *
 * This source file is subject to the new BSD license that is bundled
 * with this package in the file LICENSE.txt.
 * It is also available through the world-wide-web at this URL:
 * http://framework.zend.com/license/new-bsd
 * If you did not receive a copy of the license and are unable to
 * obtain it through the world-wide-web, please send an email
 * to license@zend.com so we can send you a copy immediately.
 *
 * @category   Zend
 * @package    Zend_Db
 * @subpackage Statement
 * @copyright  Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
 * @license    http://framework.zend.com/license/new-bsd     New BSD License
 * @version    $Id: Mysqli.php 18951 2009-11-12 16:26:19Z alexander $
 */


/**
 * @see Zend_Db_Statement
 */
require_once 'Zend/Db/Statement.php';


/**
 * Extends for Mysqli
 *
 * @category   Zend
 * @package    Zend_Db
 * @subpackage Statement
 * @copyright  Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
 * @license    http://framework.zend.com/license/new-bsd     New BSD License
 */
class Zend_Db_Statement_Mysqli extends Zend_Db_Statement
{

	/**
	 * Column names.
	 *
	 * @var array
	 */
	protected $_keys;

	/**
	 * Fetched result values.
	 *
	 * @var array
	 */
	protected $_values;

	/**
	 * @var array
	 */
	protected $_meta = null;

	/**
	 * @param  string $sql
	 * @return void
	 * @throws Zend_Db_Statement_Mysqli_Exception
	 */
	public function _prepare($sql)
	{
		$mysqli = $this->_adapter->getConnection();

		$this->_stmt = $mysqli->prepare($sql);

		if ($this->_stmt === false || $mysqli->errno) {
			/**
			 * @see Zend_Db_Statement_Mysqli_Exception
			 */
			require_once 'Zend/Db/Statement/Mysqli/Exception.php';
			throw new Zend_Db_Statement_Mysqli_Exception("Mysqli prepare error: " . $mysqli->error, $mysqli->errno);
		}
	}

	/**
	 * Binds a parameter to the specified variable name.
	 *
	 * @param mixed $parameter Name the parameter, either integer or string.
	 * @param mixed $variable  Reference to PHP variable containing the value.
	 * @param mixed $type      OPTIONAL Datatype of SQL parameter.
	 * @param mixed $length    OPTIONAL Length of SQL parameter.
	 * @param mixed $options   OPTIONAL Other options.
	 * @return bool
	 * @throws Zend_Db_Statement_Mysqli_Exception
	 */
	protected function _bindParam($parameter, &$variable, $type = null, $length = null, $options = null)
	{
		return true;
	}

	/**
	 * Closes the cursor and the statement.
	 *
	 * @return bool
	 */
	public function close()
	{
		if ($this->_stmt) {
			$r = $this->_stmt->close();
			$this->_stmt = null;
			return $r;
		}
		return false;
	}

	/**
	 * Closes the cursor, allowing the statement to be executed again.
	 *
	 * @return bool
	 */
	public function closeCursor()
	{
		if ($stmt = $this->_stmt) {
			$mysqli = $this->_adapter->getConnection();
			while ($mysqli->more_results()) {
				$mysqli->next_result();
			}
			$this->_stmt->free_result();
			return $this->_stmt->reset();
		}
		return false;
	}

	/**
	 * Returns the number of columns in the result set.
	 * Returns null if the statement has no result set metadata.
	 *
	 * @return int The number of columns.
	 */
	public function columnCount()
	{
		if (isset($this->_meta) && $this->_meta) {
			return $this->_meta->field_count;
		}
		return 0;
	}

	/**
	 * Retrieves the error code, if any, associated with the last operation on
	 * the statement handle.
	 *
	 * @return string error code.
	 */
	public function errorCode()
	{
		if (!$this->_stmt) {
			return false;
		}
		return substr($this->_stmt->sqlstate, 0, 5);
	}

	/**
	 * Retrieves an array of error information, if any, associated with the
	 * last operation on the statement handle.
	 *
	 * @return array
	 */
	public function errorInfo()
	{
		if (!$this->_stmt) {
			return false;
		}
		return array(
		substr($this->_stmt->sqlstate, 0, 5),
		$this->_stmt->errno,
		$this->_stmt->error,
		);
	}

	/**
	 * Executes a prepared statement.
	 *
	 * @param array $params OPTIONAL Values to bind to parameter placeholders.
	 * @return bool
	 * @throws Zend_Db_Statement_Mysqli_Exception
	 */
	public function _execute(array $params = null)
	{
		if (!$this->_stmt) {
			return false;
		}

		// if no params were given as an argument to execute(),
		// then default to the _bindParam array
		if ($params === null) {
			$params = $this->_bindParam;
		}
		// send $params as input parameters to the statement
		if ($params) {
			array_unshift($params, str_repeat('s', count($params)));
			$stmtParams = array();
			foreach ($params as $k => &$value) {
				$stmtParams[$k] = &$value;
			}
			call_user_func_array(
			array($this->_stmt, 'bind_param'),
			$stmtParams
			);
		}

		// execute the statement
		$retval = $this->_stmt->execute();
		if ($retval === false) {
			/**
			 * @see Zend_Db_Statement_Mysqli_Exception
			 */
			require_once 'Zend/Db/Statement/Mysqli/Exception.php';
			throw new Zend_Db_Statement_Mysqli_Exception("Mysqli statement execute error : " . $this->_stmt->error, $this->_stmt->errno);
		}


		// retain metadata
		if ($this->_meta === null) {
			$this->_meta = $this->_stmt->result_metadata();
			if ($this->_stmt->errno) {
				/**
				 * @see Zend_Db_Statement_Mysqli_Exception
				 */
				require_once 'Zend/Db/Statement/Mysqli/Exception.php';
				throw new Zend_Db_Statement_Mysqli_Exception("Mysqli statement metadata error: " . $this->_stmt->error, $this->_stmt->errno);
			}
		}

		// statements that have no result set do not return metadata
		if ($this->_meta !== false) {

			// get the column names that will result
			$this->_keys = array();
			foreach ($this->_meta->fetch_fields() as $col) {
				$this->_keys[] = $this->_adapter->foldCase($col->name);
			}

			// set up a binding space for result variables
			$this->_values = array_fill(0, count($this->_keys), null);

			// set up references to the result binding space.
			// just passing $this->_values in the call_user_func_array()
			// below won't work, you need references.
			$refs = array();
			foreach ($this->_values as $i => &$f) {
				$refs[$i] = &$f;
			}

			$this->_stmt->store_result();
			// bind to the result variables
			call_user_func_array(
			array($this->_stmt, 'bind_result'),
			$this->_values
			);
		}
		return $retval;
	}


	/**
	 * Fetches a row from the result set.
	 *
	 * @param int $style  OPTIONAL Fetch mode for this fetch operation.
	 * @param int $cursor OPTIONAL Absolute, relative, or other.
	 * @param int $offset OPTIONAL Number for absolute or relative cursors.
	 * @return mixed Array, object, or scalar depending on fetch mode.
	 * @throws Zend_Db_Statement_Mysqli_Exception
	 */
	public function fetch($style = null, $cursor = null, $offset = null)
	{
		if (!$this->_stmt) {
			return false;
		}
		// fetch the next result
		$retval = $this->_stmt->fetch();
		switch ($retval) {
			case null: // end of data
			case false: // error occurred
				$this->_stmt->reset();
				return false;
			default:
				// fallthrough
		}

		// make sure we have a fetch mode
		if ($style === null) {
			$style = $this->_fetchMode;
		}

		// dereference the result values, otherwise things like fetchAll()
		// return the same values for every entry (because of the reference).
		$values = array();
		foreach ($this->_values as $key => $val) {
			$values[] = $val;
		}

		$row = false;
		switch ($style) {
			case Zend_Db::FETCH_NUM:
				$row = $values;
				break;
			case Zend_Db::FETCH_ASSOC:
				$row = array_combine($this->_keys, $values);
				break;
			case Zend_Db::FETCH_BOTH:
				$assoc = array_combine($this->_keys, $values);
				$row = array_merge($values, $assoc);
				break;
			case Zend_Db::FETCH_OBJ:
				$row = (object) array_combine($this->_keys, $values);
				break;
			case Zend_Db::FETCH_BOUND:
				$assoc = array_combine($this->_keys, $values);
				$row = array_merge($values, $assoc);
				return $this->_fetchBound($row);
				break;
			default:
				/**
				 * @see Zend_Db_Statement_Mysqli_Exception
				 */
				require_once 'Zend/Db/Statement/Mysqli/Exception.php';
				throw new Zend_Db_Statement_Mysqli_Exception("Invalid fetch mode '$style' specified");
				break;
		}
		return $row;
	}

	/**
	 * Retrieves the next rowset (result set) for a SQL statement that has
	 * multiple result sets.  An example is a stored procedure that returns
	 * the results of multiple queries.
	 *
	 * @return bool
	 * @throws Zend_Db_Statement_Mysqli_Exception
	 */
	public function nextRowset()
	{
		/**
		 * @see Zend_Db_Statement_Mysqli_Exception
		 */
		require_once 'Zend/Db/Statement/Mysqli/Exception.php';
		throw new Zend_Db_Statement_Mysqli_Exception(__FUNCTION__.'() is not implemented');
	}

	/**
	 * Returns the number of rows affected by the execution of the
	 * last INSERT, DELETE, or UPDATE statement executed by this
	 * statement object.
	 *
	 * @return int     The number of rows affected.
	 */
	public function rowCount()
	{
		if (!$this->_adapter) {
			return false;
		}
		$mysqli = $this->_adapter->getConnection();
		return $mysqli->affected_rows;
	}

}